﻿// Copyright Dirk Lemstra https://github.com/dlemstra/Magick.NET.
// Licensed under the Apache License, Version 2.0.

using System;
using System.IO;
using Xunit;

namespace Magick.NET.Tests;

internal class TestStream : Stream
{
    private readonly bool _canRead;
    private readonly bool _canSeek;
    private readonly bool _canWrite;
    private long _length;

    protected TestStream(Stream innerStream, bool canSeek)
    {
        Assert.True(innerStream.CanRead);
        Assert.True(innerStream.CanSeek);

        InnerStream = innerStream;
        _canRead = true;
        _canWrite = true;
        _canSeek = canSeek;
    }

    private TestStream(bool canRead = true, bool canWrite = true, bool canSeek = true)
    {
        _canRead = canRead;
        _canWrite = canWrite;
        _canSeek = canSeek;
    }

    public override bool CanRead
        => _canRead;

    public override bool CanSeek
        => _canSeek;

    public override bool CanWrite
        => _canWrite;

    public override long Length
        => InnerStream?.Length ?? _length;

    public override long Position
    {
        get
        {
            if (!CanRead || !CanSeek)
                throw new NotImplementedException();

            return InnerStream?.Position ?? 0;
        }
        set => throw new NotImplementedException();
    }

    protected Stream InnerStream { get; }

    public static TestStream ThatCannotRead()
        => new TestStream(canRead: false);

    public static TestStream ThatCannotSeek()
        => new TestStream(canSeek: false);

    public static TestStream ThatCannotWrite()
        => new TestStream(canWrite: false);

    public static TestStream ThatCanOnlyRead()
        => new TestStream(canWrite: false, canSeek: false);

    public override void Flush()
        => throw new NotImplementedException();

    public override int Read(byte[] buffer, int offset, int count)
        => InnerStream.Read(buffer, offset, count);

    public override long Seek(long offset, SeekOrigin origin)
        => InnerStream.Seek(offset, origin);

    public override void SetLength(long value)
        => _length = value;

    public override void Write(byte[] buffer, int offset, int count)
        => InnerStream.Write(buffer, offset, count);
}
